package com.hongyi.demo.pojo.TestList;


import java.util.ArrayList;
import java.util.Scanner;

/**
 * @author hongyi
 * @date 2024/10/7 21:03
 */
public class StudentSystem {
    public static void main(String[] args) {
        ArrayList<Student01> list = new ArrayList<>();
        while (true) {
            System.out.println("--------欢迎来到黑马学生管理系统---------");
            System.out.println("1:添加学生");
            System.out.println("2:删除学生");
            System.out.println("3:修改学生");
            System.out.println("4:查询学生");
            System.out.println("5:退出");
            System.out.println("请输入您的选择:");
            Scanner sc = new Scanner(System.in);
            System.out.println("请输入:");
            String choose = sc.next();
            switch (choose) {
                case "1" -> addStudent01(list);
                case "2" -> deleteStudent(list);
                case "3" -> updateStudent(list);
                case "4" -> queryStudent(list);
                case "5" -> {
                    System.out.println("退出");
                    //break loop;
                    System.exit(0);
                }
                default -> System.out.println("无效输入:");
            }
        }

    }

    public static void addStudent01(ArrayList<Student01> list) {
        Scanner sc = new Scanner(System.in);
        Student01 s = new Student01();
        System.out.println("请输入学生Id:");
        String id = null;
        while (true) {
            id = sc.next();
            boolean shifou = shifou(list, id);
            if (shifou) {
                System.out.println("Id已经存在请从新输入:");
            } else {
                // id 不存在
                s.setId(id);
                break;
            }
        }

        System.out.println("请输入学生姓名:");
        String name = sc.next();
        System.out.println("请输入学生年龄:");
        String age = sc.next();
        System.out.println("请输入学生家庭住址:");
        String address = sc.next();
        Student01 St = new Student01(id, name, age, address);
        list.add(St);


        System.out.println("添加学生成功");

    }

    public static void deleteStudent(ArrayList<Student01> list) {
        Scanner scanner = new Scanner(System.in);
        System.out.println("请输入你要删除的学生学号：");
        String sid = scanner.next();
        int index = - 1; // 这就就是作为一个表示
        if (list.size() == 0) {
            System.out.println("当前没有用户");
            return ;
        }

        // 遍历
          for (int i = 0; i < list.size(); i ++ ) {
              Student01 student01 = list.get(i);
              String id = student01.getId(); // 获取对象ID
              if (id.equals(sid)) {
                  index = i;
                  list.remove(i);
                  break;
              }
          }

          if (index == -1) { // 没有这个对象
              System.out.println("该对象不存在");
          } else {
              System.out.println("删除成功！");
          }
    }
    public  static void updateStudent(ArrayList<Student01> list) {
        if(list.size() == 0) {
            System.out.println("不存在");
            return ;
        }
        Scanner sc = new Scanner(System.in);
        System.out.println("输入需要修改元素的ID");
        String sid = sc.next(); // 输入需要修改元素的ID

        // 查找学生
        Student01 studentToUpdate = null;
        int count = -1; // 如果做标记的话不要用0
        for (int i = 0; i < list.size(); i ++ ) {
            Student01 student01 = list.get(i);
            String id = student01.getId();
            if (id.equals(sid)) {
                count = i; // 记录一下那个结合符合要求
               studentToUpdate = student01; // 对象进行复制
                break;
            }
        }
        if (studentToUpdate == null) {
            System.out.println("未找到ID为 " + sid + " 的学生。");
            return ;
        }
        // 获取新信息
        System.out.println("请输入学生姓名:");
        String name = sc.next();
        System.out.println("请输入学生年龄:");
        String age = sc.next();
        System.out.println("请输入学生家庭住址:");
        String address = sc.next();


       // 更新对象属性
        Student01 student011 = new Student01();
        student011.setId(sid);
        student011.setAge(age);
        student011.setAddress(address);
        student011.setName(name);
        list.set(count,student011); // 进行修改
        System.out.println("修改成功!");
    }

    public  static void queryStudent(ArrayList<Student01> list) {
        if (list.size() == 0) {
            System.out.println("当前无学生信息,请添加后在查询");
            return ;
        }
        System.out.println("学号\t姓名\t年龄\t\t居住地");
        for (int i =0; i < list.size(); i ++ ) {
            Student01 student01 = list.get(i);
           /* System.out.println(student01.toString()); // 遍历集合*/
            System.out.println(student01.getId() + "\t" + student01.getName() + "\t" + student01.getAge() + "岁\t" + student01.getAddress());
        }
    }

    public  static boolean shifou(ArrayList<Student01> list, String id) {
        for (int i =0; i < list.size(); i ++ ) {
            Student01 stun = list.get(i);  // 获取集合中的对象
            String id1 = stun.getId(); // 获取对象的ID
            if (id.equals(id1)) { // 相同
                //
                return true;
            }
        }
        return false; // 不相同
    }
}
